# -*- coding: utf-8 -*-

import os

# pip install openai
from openai import OpenAI

client = OpenAI(
    api_key=os.getenv("AI_DASHSCOPE_API_KEY"),
    base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
)
completion = client.chat.completions.create(
    model="qwen3-coder-plus",
    messages=[
        {'role': 'system', 'content': 'You are a helpful assistant.'},
        {'role': 'user',
         'content': """
             请编写一个Python函数 find_prime_numbers，该函数接受一个整数 n 作为参数，并返回一个包含所有小于 n 的质数（素数）的列表。
             质数是指仅能被1和其自身整除的正整数，如2, 3, 5, 7等。不要输出非代码的内容。
            """
         }],
)
print("=" * 20 + "回复内容" + "=" * 20)
print(completion.choices[0].message.content)
print("=" * 20 + "Token消耗" + "=" * 20)
print(completion.usage)
